#!/usr/bin/env python3
"""
数据库重置脚本 - 安全地重置数据库
"""

import os
import sys
from app import create_app
from models import db

def main():
    """主函数"""
    print("⚠️  警告：这将删除所有现有数据！")
    confirm = input("确定要重置数据库吗？(输入 'yes' 确认): ")
    
    if confirm.lower() != 'yes':
        print("操作已取消")
        return
    
    print("开始重置数据库...")
    
    # 创建应用
    app = create_app()
    
    with app.app_context():
        try:
            # 删除所有表
            print("删除所有表...")
            db.drop_all()
            
            # 重新创建所有表
            print("重新创建表...")
            db.create_all()
            
            print("数据库重置完成！")
            print("请运行 'python init_db.py' 来初始化示例数据")
            
        except Exception as e:
            print(f"数据库重置失败: {e}")
            sys.exit(1)

if __name__ == '__main__':
    main()
